1980D - GCD-sequence - CodeForces Solution


greedy implementation math math number theory number theory

Please click on ads to support us..

Python Code:

from math import gcd

def solution():
    def find(non_coprime_seq):
        curr_gcd = 1
        return next((index for index in range(len(non_coprime_seq)-1) if not curr_gcd <= (curr_gcd := gcd(non_coprime_seq[index], non_coprime_seq[index+1]))), 0)

    num_elements = int(input())
    elements = list(map(int, input().split()))
    find_index = find(elements)
    return "YES" if find_index == 0 or any(find(elements[:remove_index] + elements[remove_index+1:]) == 0 for remove_index in range(find_index-1, (find_index+1)+1)) else "NO"

for _ in range(int(input())):
    print(solution())


Comments

Submit
0 Comments
More Questions

911A - Nearest Minimums
102B - Sum of Digits
707A - Brain's Photos
1331B - Limericks
305B - Continued Fractions
1165B - Polycarp Training
1646C - Factorials and Powers of Two
596A - Wilbur and Swimming Pool
1462B - Last Year's Substring
1608B - Build the Permutation
1505A - Is it rated - 2
169A - Chores
765A - Neverending competitions
1303A - Erasing Zeroes
1005B - Delete from the Left
94A - Restoring Password
1529B - Sifid and Strange Subsequences
1455C - Ping-pong
1644C - Increase Subarray Sums
1433A - Boring Apartments
1428B - Belted Rooms
519B - A and B and Compilation Errors
1152B - Neko Performs Cat Furrier Transform
1411A - In-game Chat
119A - Epic Game
703A - Mishka and Game
1504C - Balance the Bits
988A - Diverse Team
1312B - Bogosort
1616B - Mirror in the String